package 组合问题;

import java.util.ArrayList;
import java.util.List;

/**
 * @author zhengTao
 * @version 1.0
 * @description: 组合
 * 给定两个整数 n 和 k，返回范围 [1, n] 中所有可能的 k 个数的组合。
 * 你可以按 任何顺序 返回答案。
 * @date 2022/11/12 17:36
 */
public class Demo77 {
    public static void main(String[] args) {
        List<List<Integer>> combine = combine(4, 2);
        System.out.println(combine);
    }

    static List<List<Integer>> res = new ArrayList<>();
    static List<Integer> path = new ArrayList<>();

    public static List<List<Integer>> combine(int n, int k) {
        backTrack(1,n, k);
        return res;
    }

    //回溯三部曲
    //1，回溯函数
    private static void backTrack(int start, int n, int k) {
        if (path.size() == k) {
            List<Integer> temp = new ArrayList<>(path);
            res.add(temp);
            return;
        }
        for (int i = start; i <= n; i++) {
            path.add(i);
            //index 不能使用i++
            backTrack(i+1, n, k);
            //回溯
            path.remove(path.size() - 1);
        }
    }
    //1，回溯函数优化
    private static void backTrack1(int start, int n, int k) {
        if (path.size() == k) {
            List<Integer> temp = new ArrayList<>(path);
            res.add(temp);
            return;
        }
        //剪枝操作
        // k- path.size()  剩余需要遍历元素个数
        //n-(k- path.size()) 至多开始需要遍历的元素
        for (int i = start; i <= n-(k- path.size())+1; i++) {
            path.add(i);
            //index 不能使用i++
            backTrack(i+1, n, k);
            //回溯
            path.remove(path.size() - 1);
        }
    }
}
